# 配置超参数
import torch
# GPU
# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# CPU
device = torch.device("cpu")
MAX_LEN = 512
TRAIN_BATCH_SIZE = 8
VALID_BATCH_SIZE = 4
LEARNING_RATE = 1e-05

model_dir = "./finetune_models/my_finetune_model.pth"

output_vocab_file = './models/'
